TypeError: 'module' object is not callable - while using UMAP